def print_story(text):
    """打印故事内容"""
    print("\n" + text)

def main():
    print("===== 绿野仙踪简易冒险 =====")
    print("你将扮演多萝西，在奥兹国展开冒险之旅！\n")
    
    # 角色状态
    companions = []
    
    # 场景1：堪萨斯州
    print_story("龙卷风把你的房子卷到了一个陌生的地方。你走出房子，发现自己来到了矮人国。")
    input("按回车继续...")
    
    # 场景2：矮人国
    print_story("北方好女巫告诉你：'沿着黄砖路去翡翠城，奥兹巫师能帮你回家。'")
    input("按回车开始旅程...")
    
    # 场景3：遇到稻草人
    print_story("你在路上看到一个稻草人挂在杆子上。他说：'请帮我下来，我想找巫师要一个大脑。'")
    choice = input("要帮助稻草人吗？(y/n): ").lower()
    if choice == 'y':
        companions.append("稻草人")
        print_story("稻草人加入了你的队伍！")
    input("按回车继续...")
    
    # 场景4：遇到铁皮人
    print_story("你在森林里遇到一个生锈的铁皮人。他说：'请帮我上油，我想找巫师要一颗心。'")
    choice = input("要帮助铁皮人吗？(y/n): ").lower()
    if choice == 'y':
        companions.append("铁皮人")
        print_story("铁皮人加入了你的队伍！")
    input("按回车继续...")
    
    # 场景5：遇到胆小狮
    print_story("一只狮子跳出来挡住了你的路，但他其实很胆小。他说：'我想找巫师要勇气。'")
    choice = input("要让狮子加入吗？(y/n): ").lower()
    if choice == 'y':
        companions.append("胆小狮")
        print_story("胆小狮加入了你的队伍！")
    input("按回车继续...")
    
    # 场景6：翡翠城
    print_story("你们终于到达了翡翠城。巫师说：'先帮我打败西方女巫，我就满足你们的愿望。'")
    input("按回车接受任务...")
    
    # 场景7：对抗西方女巫
    print_story("你们找到了西方女巫的城堡。")
    if len(companions) >= 2:
        print_story("在同伴们的帮助下，你们成功打败了西方女巫！")
    else:
        print_story("虽然艰难，但你还是设法打败了西方女巫！")
    input("按回车返回翡翠城...")
    
    # 场景8：结局
    print_story("\n===== 结局 =====")
    print(f"巫师帮助了你的同伴：")
    if "稻草人" in companions:
        print("- 稻草人发现自己其实很聪明")
    if "铁皮人" in companions:
        print("- 铁皮人发现自己其实有爱心")
    if "胆小狮" in companions:
        print("- 胆小狮发现自己其实很勇敢")
    
    print("\n北方好女巫告诉你，你的银鞋可以带你回家。")
    print("你敲了敲鞋跟，许愿回家...")
    print("\n你回到了堪萨斯州的家！冒险结束了。")

if __name__ == "__main__":
    main()
    